package com.nzd.slsharing.controller;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import com.nzd.slsharing.util.Result;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

@RestController
@RequestMapping("/api/file")
public class FileController {
    /**
     * 上传图片接口
     */
    @PostMapping("/upload")
    public Result upload(MultipartFile file, HttpServletRequest request) {
        // 定义支持的图片类型
        List<String> allowedImageTypes = Arrays.asList("jpg", "jpeg", "png", "gif", "bmp");

        try {
            // 获取文件的原始文件名
            String originalFilename = file.getOriginalFilename();
            // 获取文件的扩展名
            String type = FileUtil.extName(originalFilename).toLowerCase();  // 转小写，避免大小写不一致问题

            // 判断文件类型是否在允许的图片类型列表中
            if (!allowedImageTypes.contains(type)) {
                return Result.fail("只允许上传图片类型文件");
            }

            // 随机生成一个新的文件名
            String uuid = IdUtil.fastSimpleUUID();
            String fileUUID = uuid + StrUtil.DOT + type;

            // 自定义图片存储路径
            String path = "E:/upload/image/";
            SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy/MM/dd/");
            String dir = simpleDateFormat.format(new Date());

            // 创建图片存储目录
            File newFileMkr = new File(path + dir);
            if (!newFileMkr.exists()) {
                newFileMkr.mkdirs();
            }

            // 创建新文件
            File newFile = new File(path + dir + fileUUID);
            // 保存文件
            file.transferTo(newFile);

            // 构造图片访问的URL
            String url = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/image/" + dir + fileUUID;

            // 返回图片访问的URL
            return Result.ok(url);
        } catch (Exception e) {
            e.printStackTrace();
            // 处理异常
            return Result.fail("上传失败");
        }
    }
}
